<template>
  <div class="excel-box">
    <div class="btn-box">
      <el-button
        :loading="loading"
        style="margin-left: 16px"
        type="primary"
        @click="handleUpload"
      >
        点击上传
      </el-button>
    </div>

    <input
      ref="excel-upload-input"
      class="excel-upload-input"
      type="file"
      accept=".xlsx, .xls"
      @change="handleClick"
    />
    <div
      class="drop"
      @drop="handleDrop"
      @dragover="handleDragover"
      @dragenter="handleDragover"
    >
      <i class="el-icon-upload"></i>
      <span>将文件拖到此处</span>
    </div>
  </div>
</template>

<script>
import XLSX from 'xlsx'

export default {
  props: {
    beforeUpload: Function, // eslint-disable-line
    onSuccess: Function // eslint-disable-line
  },
  data() {
    return {
      loading: false,
      excelData: {
        header: null,
        results: null
      }
    }
  },
  methods: {
    generateData({ header, results }) {
      this.excelData.header = header
      this.excelData.results = results

      // 这句话相当于是
      /*
        if (this.onSuccess) {
          this.onSuccess(this.excelData)
        }
      */
      this.onSuccess && this.onSuccess(this.excelData)
    },
    handleDrop(e) {
      e.stopPropagation()
      e.preventDefault()
      if (this.loading) return
      const files = e.dataTransfer.files
      if (files.length !== 1) {
        this.$message.error('Only support uploading one file!')
        return
      }
      const rawFile = files[0] // only use files[0]

      if (!this.isExcel(rawFile)) {
        this.$message.error(
          'Only supports upload .xlsx, .xls, .csv suffix files'
        )
        return false
      }
      this.upload(rawFile)
      e.stopPropagation()
      e.preventDefault()
    },
    handleDragover(e) {
      e.stopPropagation()
      e.preventDefault()
      e.dataTransfer.dropEffect = 'copy'
    },

    // 按钮的点击事件
    handleUpload() {
      // 调用了 input type=file 的点击事件
      this.$refs['excel-upload-input'].click()
    },

    // 选择完文件调用的事件
    handleClick(e) {
      // 拿到选择的文件，是一个数组，因为有可能选择多个文件
      const files = e.target.files
      // 我们只能选1个文件，所以直接取下标0得到这个文件
      const rawFile = files[0] // only use files[0]
      // 如果没有文件直接return不往下执行
      if (!rawFile) return
      // 有文件，又调用了下面upLoad方法，并传入文件
      this.upload(rawFile)
    },
    upload(rawFile) {
      // 主要是为了防止bug，把file清空，可以方便下次继续选择文件
      // 原因：file如果两次选择同一个文件第二次不会触发，所以把value设置为空
      // 哪怕两次选择的都是同一个文件也会触发
      this.$refs['excel-upload-input'].value = null // fix can't select the same excel

      // 如果没有传beforeUpload钩子
      if (!this.beforeUpload) {
        // 直接调用下面的readerData方法，传入本次选择的文件
        this.readerData(rawFile)
        return
      }

      // 能来到这证明有beforeUpload钩子
      // 调用了这个钩子，并且把文件传递过去了
      // 得到return的结果
      const before = this.beforeUpload(rawFile)
      if (before) {
        this.readerData(rawFile)
      }
    },

    // 真正读取excel的代码
    readerData(rawFile) {
      // 开启加载动画
      this.loading = true
      return new Promise((resolve, reject) => {
        // 实例化一个文件读取器
        // 你要读取文件的内容就要有这个对象
        const reader = new FileReader()

        // 读取完会调用的事件
        reader.onload = (e) => {
          // e.target.result就是读取到的内容（二进制数组）
          const data = e.target.result

          // 调用XLSX的插件的read方法，传入这个二进制数组
          // workbook就是XLSX根据二进制转换的excel内容
          const workbook = XLSX.read(data, { type: 'array' })
          // 这是获取excel的第一张表的名字
          const firstSheetName = workbook.SheetNames[0]
          // 读取某张表的内容，所以我这里传入了第一张表
          // 意味着读取出第一张表里的内容
          const worksheet = workbook.Sheets[firstSheetName]
          // 读取出这张表里的表头（第一行）
          const header = this.getHeaderRow(worksheet)
          // 读取出这张表里的内容
          const results = XLSX.utils.sheet_to_json(worksheet)
          // 然后调用当前组件里的generateData方法，传入了一个对象
          // 对象里有本次读取到的header和results
          this.generateData({ header, results })
          // 读取完了把加载动画关闭
          this.loading = false
          // 读取成功调用resolve
          resolve()
        }

        // 利用文件读取对象传入了文件
        // 这就是在读取这个文件，并且读取成二进制数组
        // 当他读取完毕就会调用上面代码里的onLoad事件
        reader.readAsArrayBuffer(rawFile)
      })
    },
    // 这个方法就是读取出表头的方法
    getHeaderRow(sheet) {
      // 声明一个放表头的数组
      const headers = []
      // 调用插件方法得到范围
      const range = XLSX.utils.decode_range(sheet['!ref'])
      let C
      const R = range.s.r
      /* start in the first row */
      for (C = range.s.c; C <= range.e.c; ++C) {
        /* walk every column in the range */
        const cell = sheet[XLSX.utils.encode_cell({ c: C, r: R })]
        /* find the cell in the first row */
        let hdr = 'UNKNOWN ' + C // <-- replace with your desired default
        if (cell && cell.t) hdr = XLSX.utils.format_cell(cell)
        headers.push(hdr)
      }
      return headers
    },
    isExcel(file) {
      return /\.(xlsx|xls|csv)$/.test(file.name)
    }
  }
}
</script>

<style lang="scss" scoped>
.excel-box {
  display: flex;
  height: 100vh;
  justify-content: center;
  align-items: center;

  .drop,
  .btn-box {
    width: 350px;
    height: 200px;
    border: 1px dashed #bbb;
    display: flex;
    justify-content: center;
    align-items: center;
  }

  .excel-upload-input {
    display: none;
    z-index: -9999;
  }
  .drop {
    flex-direction: column;
    font-size: 24px;
    border-radius: 5px;
    text-align: center;
    color: #bbb;
    i {
        font-size: 80px;
    }
  }
}
</style>
